package com.poi.demo.excel;

import org.apache.poi.hssf.usermodel.HSSFDataFormat;
import org.apache.poi.ss.usermodel.Cell;
import org.apache.poi.ss.usermodel.CellStyle;
import org.apache.poi.ss.usermodel.CellType;
import org.apache.poi.ss.usermodel.Workbook;
import org.apache.poi.xssf.usermodel.XSSFWorkbook;

import java.io.InputStream;

/**
 * 在开发中我们经常使用HSSF用来操作Excel处理表格数据，对于其它的不经常使用。
 * <p>
 * HSSF 是Horrible SpreadSheet Format的缩写，通过HSSF，你可以用纯Java代码来读取、写入、
 * 修改Excel文件。
 * HSSF 为读取操作提供了两类API：usermodel和eventusermodel，即“用户模型”和“事件-用户模型”。
 * <p>
 * 常用的类和方法
 * HSSFWorkbook ：工作簿，代表一个excel的整个文档
 * HSSFWorkbook(); // 创建一个新的工作簿
 * HSSFWorkbook(InputStream inputStream); // 创建一个关联输入流的工作簿，可以将一个excel文件封装成工作簿
 * HSSFSheet createSheet(String sheetname); 创建一个新的Sheet
 * HSSFSheet getSheet(String sheetName); 通过名称获取Sheet
 * HSSFSheet getSheetAt(int index); // 通过索引获取Sheet，索引从0开始
 * HSSFCellStyle createCellStyle()； 创建单元格样式
 * int getNumberOfSheets(); 获取sheet的个数
 * setActiveSheet(int index); 设置默认选中的工作表
 * write();
 * write(File newFile);
 * write(OutputStream stream);
 * HSSFSheet：工作表
 * HSSFRow createRow(int rownum); 创建新行，需要指定行号，行号从0开始
 * HSSFRow getRow(int index); 根据索引获取指定的行
 * int addMergedRegion(CellRangeAddress region); 合并单元格
 * CellRangeAddress(int firstRow, int lastRow, int firstCol, int lastCol); 单元格范围, 用于合并单元格，需要指定要合并的首行、最后一行、首列、最后一列。
 * autoSizeColumn(int column); 自动调整列的宽度来适应内容
 * getLastRowNum(); 获取最后的行的索引，没有行或者只有一行的时候返回0
 * setColumnWidth(int columnIndex, int width); 设置某一列的宽度，width=字符个数 * 256，例如20个字符的宽度就是20 * 256
 * <p>
 * HSSFRow ：行
 * HSSFCell createCell(int column); 创建新的单元格
 * HSSFCell setCell(shot index);
 * HSSFCell getCell(shot index);
 * setRowStyle(HSSFCellStyle style); 设置行样式
 * short getLastCellNum(); 获取最后的单元格号，如果单元格有第一个开始算，lastCellNum就是列的个数
 * setHeightInPoints(float height); 设置行的高度
 * <p>
 * HSSFCell：单元格
 * setCellValue(String value); 设置单元格的值
 * setCellType(); 设置单元格类型，如 字符串、数字、布尔等
 * setCellStyle(); 设置单元格样式
 * String getStringCellValue(); 获取单元格中的字符串值
 * setCellStyle(HSSFCellStyle style); 设置单元格样式，例如字体、加粗、格式化
 * setCellFormula(String formula); 设置计算公式，计算的结果作为单元格的值，也提供了异常常用的函数，如求和”sum(A1,C1)”、日期函数、字符串相关函数、CountIf和SumIf函数、随机数函数等
 * <p>
 * HSSFCellStyle ：单元格样式
 * setFont(Font font); 为单元格设置字体样式
 * setAlignment(HorizontalAlignment align); // 设置水平对齐方式
 * setVerticalAlignment(VerticalAlignment align); // 设置垂直对齐方式
 * setFillPattern(FillPatternType fp);
 * setFillForegroundColor(short bg); 设置前景色
 * setFillBackgroundColor(short bg); 设置背景颜色
 * <p>
 * HSSFFont：字体，
 * setColor(short color); // 设置字体颜色
 * setBold(boolean bold); // 设置是否粗体
 * setItalic(boolean italic); 设置倾斜
 * setUnderline(byte underline); 设置下划线
 * <p>
 * HSSFName：名称
 * HSSFDataFormat ：日期格式化
 * HSSFHeader ： Sheet的头部
 * HSSFFooter ：Sheet的尾部
 * HSSFDateUtil ：日期工具
 * HSSFPrintSetup ：打印设置
 * HSSFErrorConstants：错误信息表
 * <p>
 * Excel中的工作簿、工作表、行、单元格中的关系：
 * <p>
 * 一个Excel文件对应于一个workbook(HSSFWorkbook)，
 * 一个workbook可以有多个sheet（HSSFSheet）组成，
 * 一个sheet是由多个row（HSSFRow）组成，
 * 一个row是由多个cell（HSSFCell）组成
 * <p>
 * 王永吉
 */
public class ExcelKit {
    /**
     * 获取工作簿对象(Excel对象)
     *
     * @param fileName        文件名
     * @param fileInputStream 文件输入流
     * @return 工作簿对象(Excel对象)
     */
    public static Workbook workBook(String fileName, InputStream fileInputStream) throws Exception {
        //2007
        if (fileName.endsWith("xlsx")) {
            return new XSSFWorkbook(fileInputStream);
        }
        //2003
        if (fileName.endsWith("xls")) {
            return new XSSFWorkbook(fileInputStream);
        }
        throw new Exception("获取工作簿对象失败：无法根据文件名" + fileName + ",创建工作簿对象");
    }


    public static Object cellValue(Cell cell) {
        if (cell == null) return "";

        CellType cellType = cell.getCellType();
        Object cellValue = "";
        if (cellType == CellType.NUMERIC) {
            cellValue = cell.getNumericCellValue();
        }
        if (cellType == CellType.STRING) {
            cellValue = cell.getStringCellValue();
        }
        if (cellType == CellType.FORMULA) {
            cellValue = cell.getStringCellValue();
        }
        if (cellType == CellType.BLANK) {
            cellValue = "";
        }
        if (cellType == CellType.BOOLEAN) {
            cellValue = cell.getBooleanCellValue();
        }
        if (cellType == CellType.ERROR) {
            cellValue = cell.getErrorCellValue();
        }
        return cellValue;
    }

    /**
     * 设置整列 为文本格式
     *
     * @param workbook 工作簿对象
     * @param sheet    sheet（从0开始）
     * @param column   列（从0开始）
     */
    public static void setStyleStr(Workbook workbook, int sheet, int column) {
        workbook.getSheetAt(sheet).setDefaultColumnStyle(column, styleString(workbook));
    }

    /**
     * 设置某行某列 为文本格式
     *
     * @param workbook 工作簿对象
     * @param sheet    sheet（从0开始）
     * @param row      行（从0开始）
     * @param column   列（从0开始）
     */
    public static void setStyleStr(Workbook workbook, int sheet, int row, int column) {
        workbook.getSheetAt(sheet).getRow(row).getCell(column).setCellStyle(styleString(workbook));
    }


    /**
     * 设置某行 设置为两位小数
     * 参数和 setStyleStr 一样
     */
    public static void setStyleTwoDecimal(Workbook workbook, int sheet, int column) {
        workbook.getSheetAt(sheet).setDefaultColumnStyle(column, styleTwoDecimal(workbook));
    }

    /**
     * 设置某行某列 设置为两位小数
     * 参数和 setStyleStr 一样
     */
    public static void setStyleTwoDecimal(Workbook workbook, int sheet, int row, int column) {
        workbook.getSheetAt(sheet).getRow(row).getCell(column).setCellStyle(styleTwoDecimal(workbook));
    }


    //文本格式
    private static CellStyle styleString(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setDataFormat(workbook.createDataFormat().getFormat("@"));
        return style;
    }

    //保留两位小数格式
    //用的是HSSFDataFormat.getBuiltinFormat()方法，之所以用这个，
    // 是因为0.00是Excel内嵌的格式
    private static CellStyle styleTwoDecimal(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setDataFormat(HSSFDataFormat.getBuiltinFormat("0.00"));
        return style;
    }


    //    货币格式
    public static CellStyle styleCurrency(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setDataFormat(workbook.createDataFormat().getFormat("¥#,##0"));
        return style;
    }

    //    百分比格式
    public static CellStyle stylePercentage(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setDataFormat(HSSFDataFormat.getBuiltinFormat("0.00%"));
        return style;
    }

    //    中文大写格式
    public static CellStyle styleChineseUppercase(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setDataFormat(workbook.createDataFormat().getFormat("[DbNum2][$-804]0"));
        return style;
    }

    //    科学计数法格式
    public static CellStyle styleScientificCounting(Workbook workbook) {
        CellStyle style = workbook.createCellStyle();
        style.setDataFormat(workbook.createDataFormat().getFormat("0.00E+00"));
        return style;
    }


}
